// Copyright 2016-2101 Pica.
package com.arceus.http;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;
import java.net.*;
import java.util.List;
import java.util.Map;

/**
 * @ClassName HttpClient
 * @Author iron
 * @Date 2019/9/21 17:25
 * @ModifyDate 2019/9/21 17:25
 * @Version v.1.0
 */
public class HttpClient {
    final static Logger logger = LoggerFactory.getLogger(HttpClient.class);

    private static final String FORM = "application/x-www-form-urlencoded";
    private static final String JSON = "application/json";


    private static String sendGet(String url, String param) {
        return sendGet(url, param, null);
    }

    /**
     * 向指定 URL 发送POST方法的请求
     *
     * @param url   发送请求的 URL
     * @param param 请求参数，请求参数应该是 name1=value1&name2=value2 的形式。
     * @return 所代表远程资源的响应结果
     */
    public static String sendPostFrom(String url, String param) {
        return sendPost(url, param, null, FORM);
    }

    /**
     * 向指定 URL 发送POST方法的请求
     *
     * @param url   发送请求的 URL
     * @param param 请求参数，请求参数应该是 name1=value1&name2=value2 的形式。
     * @return 所代表远程资源的响应结果
     */
    public static String sendPostJson(String url, String param) {
        return sendPost(url, param, null, JSON);
    }


    /**
     * 发送get请求到url，可以携带自定义请求头
     *
     * @param url
     * @param param
     * @param headers
     * @return
     */
    public static String sendGet(String url, String param, Map<String, String> headers) {
        String result = "";
        BufferedReader in = null;
        try {
            String urlNameString = url + "?" + param;
            URL realUrl = new URL(urlNameString);
            // 打开和URL之间的连接
            URLConnection conn = realUrl.openConnection();
            // 设置通用的请求属性
            conn.setRequestProperty("accept", "*/*");
            conn.setRequestProperty("connection", "Keep-Alive");
            conn.setRequestProperty("user-agent",
                    "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)");
            // 添加自定义的请求头
            if (headers != null) {
                for (Map.Entry<String, String> header : headers.entrySet()) {
                    conn.setRequestProperty(header.getKey(), header.getValue());
                }
            }
            // 建立实际的连接
            conn.connect();
            // 获取所有响应头字段
            Map<String, List<String>> map = conn.getHeaderFields();
            // 遍历所有的响应头字段
            for (String key : map.keySet()) {
                logger.debug("key:{},value,{}", key, map.get(key));
            }
            // 定义 BufferedReader输入流来读取URL的响应
            in = new BufferedReader(new InputStreamReader(conn.getInputStream()));
            String line;
            while ((line = in.readLine()) != null) {
                result += line;
            }
        } catch (Exception e) {
            logger.error("发送GET请求出现异常", e);
        }
        // 使用finally块来关闭输入流
        finally {
            try {
                if (in != null) {
                    in.close();
                }
            } catch (Exception e2) {
                logger.error("关闭get连接出现异常", e2);
            }
        }
        return result;
    }

    /**
     * 向指定 URL 发送POST方法的请求
     *
     * @param url     发送请求的 URL
     * @param json    请求参数，请求参数应该是 name1=value1&name2=value2 的形式。
     * @param headers 请求头信息
     * @return 所代表远程资源的响应结果
     */
    public static String sendPost(String url, String json, Map<String, String> headers, String ContentType) {
        OutputStreamWriter osw = null;
        BufferedReader in = null;
        StringBuffer buffer = new StringBuffer();
        try {
            URL realUrl = new URL(url);
            // 打开和URL之间的连接
            HttpURLConnection conn = (HttpURLConnection) realUrl.openConnection();
            // 设置通用的请求属性
            conn.setRequestProperty("accept", "*/*");
            conn.setRequestProperty("connection", "Keep-Alive");
            conn.setRequestProperty("user-agent",
                    "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)");
            conn.setRequestProperty("Content-type", ContentType);
            conn.setRequestMethod("POST");
            // 添加自定义的请求头
            if (headers != null) {
                for (Map.Entry<String, String> header : headers.entrySet()) {
                    conn.setRequestProperty(header.getKey(), header.getValue());
                }
            }
            // 发送POST请求必须设置如下两行
            conn.setDoOutput(true);
            conn.setDoInput(true);
            conn.setUseCaches(false);
            conn.connect();
            osw = new OutputStreamWriter(conn.getOutputStream(), "UTF-8");
            if (json != null) {
                osw.write(json);
            }
            // flush输出流的缓冲
            osw.flush();
            // 定义BufferedReader输入流来读取URL的响应
            String temp;
            in = new BufferedReader(new InputStreamReader(conn.getInputStream(), "UTF-8"));
            while ((temp = in.readLine()) != null) {
                buffer.append(temp);
                buffer.append("\n");
            }
        } catch (Exception e) {
            logger.error("发送 POST 请求出现异常！", e);
        }
        //使用finally块来关闭输出流、输入流
        finally {
            try {
                if (osw != null) {
                    osw.close();
                }
                if (in != null) {
                    in.close();
                }
            } catch (IOException ex) {
                logger.error("关闭连接异常", ex);
            }
        }
        return buffer.toString();
    }


    public static void main(String[] args) throws IOException {
        HttpClient.sendGet("http://www.baidu.com", null);
        sendPostJson("http://www.baidu.com", null);
    }
}